package com.cu.langchain4j.tools;

import dev.langchain4j.agent.tool.P;
import dev.langchain4j.agent.tool.Tool;

public class Relation {
    @Tool(name = "任务关系查看", value = "根据2个人物,获取2者关系")
    public String getRelation(@P("人物1") String name1, @P("人物2") String name2) {
        //查找数据库中是否包含对应的预约记录
        if((name1.equals("小摘摘") || name1.equals("丁梦平")) && (name2.equals("金天") || name2.equals("小老头"))){
            return "夫妻关系";
        }
        if((name2.equals("小摘摘") || name2.equals("丁梦平")) && (name1.equals("金天") || name1.equals("小老头"))){
            return "夫妻关系";
        }
        return "无关系";
    }
}
